from Finance_fun.base.get_driver import GetDriver
from Finance_fun.page.page05_my_investment import PageMyInvestment
import unittest

class MyInvestmentTest(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.driver = GetDriver.get_driver()
        cls.my_tz = PageMyInvestment(cls.driver)
        cls.my_tz.page_click_login_link()
        cls.my_tz.page_login("13129628532","W123456")


    @classmethod
    def tearDownClass(cls):
        GetDriver.quit_driver()

    def test01_my_tz(self):
        # 点击我的投资
        self.my_tz.my_tz()
        self.my_tz.filter("2024-05-01","2024-05-20")

    def test02_my_tz(self):
        # 点击投标中
        self.my_tz.tbz()
        self.my_tz.filter("2024-05-01","2024-05-20")

    def test03_my_tz(self):
        # 点击已结清
        self.my_tz.yjq()
        self.my_tz.filter("2024-05-01","2024-05-20")

    def test04_my_tz(self):
        # 点击已流标
        self.my_tz.ylb()
        self.my_tz.filter("2024-05-01","2024-05-20")